package com.boot2.core.utils;

import lombok.extern.apachecommons.CommonsLog;

import java.math.BigDecimal;

/**
 * @author zhangweilin
 * @date 2020/8/4 11:37
 * @Description: BigDecimal工具
 */
@CommonsLog
public class BigDecimalUtils {
    public static double add(double a, double b) {
        BigDecimal a2 = new BigDecimal(Double.toString(a));
        BigDecimal b2 = new BigDecimal(Double.toString(b));
        return a2.add(b2).doubleValue();
    }

    public static double sub(double a, double b) {
        BigDecimal a1 = new BigDecimal(Double.toString(a));
        BigDecimal b1 = new BigDecimal(Double.toString(b));
        return a1.subtract(b1).doubleValue();
    }

    public static double multiply(double a, double b) {
        BigDecimal a1 = new BigDecimal(Double.toString(a));
        BigDecimal b1 = new BigDecimal(Double.toString(b));
        return a1.multiply(b1).doubleValue();
    }

    public static BigDecimal add(BigDecimal a, BigDecimal b) {
        return a.add(b);
    }

    public static BigDecimal sub(BigDecimal a, BigDecimal b) {
        return a.subtract(b);
    }

    public static BigDecimal multiply(BigDecimal a, BigDecimal b) {
        return a.multiply(b);
    }

    // 默认除法运算精度
    private static final int DEF_DIV_SCALE = 10;

    /**
     * 提供（相对）精确的除法运算，当发生除不尽的情况时，精确到 小数点以后10位，以后的数字四舍五入。
     *
     * @param a 被除数
     * @param b 除数
     * @return 两个参数的商
     */
    public static double div(double a, double b) {
        return div(a, b, DEF_DIV_SCALE);
    }

    /**
     * 提供（相对）精确的除法运算。当发生除不尽的情况时，由scale参数指 定精度，以后的数字四舍五入。
     *
     * @param a     被除数
     * @param b     除数
     * @param scale 表示表示需要精确到小数点以后几位。
     * @return 两个参数的商
     */
    public static double div(double a, double b, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal a1 = new BigDecimal(Double.toString(a));
        BigDecimal b1 = new BigDecimal(Double.toString(b));
        return a1.divide(b1, scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供（相对）精确的除法运算。当发生除不尽的情况时，由scale参数指 定精度，以后的数字四舍五入。
     *
     * @param a     被除数
     * @param b     除数
     * @param scale 表示表示需要精确到小数点以后几位。
     * @return 两个参数的商
     */
    public static BigDecimal div(BigDecimal a, BigDecimal b, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        return a.divide(b, scale, BigDecimal.ROUND_HALF_UP);
    }

    /**
     * 提供精确的小数位四舍五入处理。
     *
     * @param a     需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static double round(double a, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal a1 = new BigDecimal(Double.toString(a));
        BigDecimal one = new BigDecimal("1");
        return a1.divide(one, scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供精确的小数位四舍五入处理。
     *
     * @param a     需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static BigDecimal round(BigDecimal a, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal one = new BigDecimal("1");
        return a.divide(one, scale, BigDecimal.ROUND_HALF_UP);
    }

    /**
     * @param obj
     * @return
     */
    public static BigDecimal toBigDecimal(Object obj) {

        BigDecimal bigDecimal = null;

        if (obj instanceof Short) {
            bigDecimal = new BigDecimal((Short) obj);

        } else if (obj instanceof Integer) {
            bigDecimal = new BigDecimal((Integer) obj);

        } else if (obj instanceof Long) {
            bigDecimal = new BigDecimal((Long) obj);

        } else if (obj instanceof Float) {
            bigDecimal = new BigDecimal((Float) obj);

        } else if (obj instanceof Double) {
            bigDecimal = new BigDecimal((Double) obj);

        } else if (obj instanceof String) {
            bigDecimal = new BigDecimal(Double.parseDouble((String) obj));

        } else if (obj instanceof BigDecimal) {
            bigDecimal = (BigDecimal) obj;

        } else {
            log.warn("to BigDecimal fail, obj type: " + obj != null ? obj.getClass() : "obj is null");
        }

        return bigDecimal;
    }
}
